/**
 * 版权声明： 版权所有 违者必究 2012
 * 日    期：12-8-3
 */
package com.rop.request;

import org.apache.commons.codec.binary.Base64;
import org.apache.commons.lang3.StringUtils;

/**
 * <pre>
 *  bbh-rop的上传文件编码格式为：
 *   fileType@BASE64编码的文件内容
 * </pre>
 *
 * @author tancheng
 * @version 1.0
 */
public class UploadFileUtils {

    public static final char SPERATOR = '@';
    public static final String FILE_PREFIX = "data:";
    /**
     * 获取文件的类型
     *
     * @param encodeFile
     * @return
     */
    public static final String getFileType(String encodeFile) {
        int speratorIndex = encodeFile.indexOf(SPERATOR);
        if (speratorIndex > -1) {
            String fileType = encodeFile.substring(0, speratorIndex);
            return fileType.toLowerCase();
        }else if(StringUtils.startsWith(encodeFile,FILE_PREFIX)) {
            String[] result=StringUtils.split(encodeFile,":|;|,");
            if(result.length>=3){
                return StringUtils.replace(result[1],"image/","").toLowerCase();
            }else{
                throw new IllegalUploadFileFormatException("文件格式不对，正确格式为：data:type;encode,content...");
            }
        } else {
            throw new IllegalUploadFileFormatException("文件格式不对，正确格式为：<文件格式>@<文件内容>");
        }
    }

    /**
     * 获取文件的字节数组
     *
     * @param encodeFile
     * @return
     */
    public static final byte[] decode(String encodeFile) {
        int speratorIndex = encodeFile.indexOf(SPERATOR);
        if (speratorIndex > -1) {
            String content = encodeFile.substring(speratorIndex + 1);
            return Base64.decodeBase64(content);
        }else if(StringUtils.startsWith(encodeFile,FILE_PREFIX)) {
            String[] result=StringUtils.split(encodeFile,":|;|,");
            if(result.length>=3){
                return  Base64.decodeBase64(result[3]);
            }else{
                throw new IllegalUploadFileFormatException("文件格式不对，正确格式为：data:type;encode,content...");
            }
        } else {
            throw new IllegalUploadFileFormatException("文件格式不对，正确格式为：<文件格式>@<文件内容>");
        }
    }

    /**
     * 将文件编码为BASE64的字符串
     *
     * @param bytes
     * @return
     */
    public static final String encode(byte[] bytes) {
        return Base64.encodeBase64String(bytes);
    }

    /**
     * 将文件编码为一个字符串
     * @param uploadFile
     * @return
     */
    public static final String encode(UploadFile uploadFile){
        StringBuilder sb = new StringBuilder();
        sb.append(uploadFile.getFileType());
        sb.append(SPERATOR);
        sb.append(encode(uploadFile.getContent()));
        return sb.toString();
    }
}

